// ========================================================================= //

// includes
#include "stdio.h"
#include "stdlib.h"
#include "time.h"
#include "string.h"
#include "limits.h"
#include "float.h"



#define JOTAI_NUM_RANDS_ 25

const unsigned rand_primes[JOTAI_NUM_RANDS_] = {179, 103, 479, 647, 229, 37, 271, 557, 263, 607, 18743, 50359, 21929, 48757, 98179, 12907, 52937, 64579, 49957, 52567, 507163, 149939, 412157, 680861, 757751};

int next_i() {
  int counter = 0;
  return rand_primes[(++counter)%JOTAI_NUM_RANDS_];
}

float next_f() {
  int counter = 0;
  return rand_primes[(++counter)%JOTAI_NUM_RANDS_] / 757751.0F;
} 


// Usage menu
void usage() {
    printf("%s", "Usage:\n\
    prog [ARGS]\n\
\nARGS:\n\
       0            big-arr\n\
       1            big-arr-10x\n\
       2            empty\n\
\n\
");

}


// ------------------------------------------------------------------------- //

#define NULL ((void*)0)
typedef unsigned long size_t;  // Customize by platform.
typedef long intptr_t; typedef unsigned long uintptr_t;
typedef long scalar_t__;  // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */
struct usb_iso_packet_descriptor {scalar_t__ status; unsigned int actual_length; int offset; } ;
struct urb {scalar_t__ status; size_t number_of_packets; unsigned char* transfer_buffer; struct usb_iso_packet_descriptor* iso_frame_desc; } ;

/* Variables and functions */

__attribute__((used)) static int st5481_isoc_flatten(struct urb *urb)
{
	struct usb_iso_packet_descriptor *pipd, *pend;
	unsigned char *src, *dst;
	unsigned int len;

	if (urb->status < 0) {
		return urb->status;
	}
	for (pipd = &urb->iso_frame_desc[0],
		     pend = &urb->iso_frame_desc[urb->number_of_packets],
		     dst = urb->transfer_buffer;
	     pipd < pend;
	     pipd++) {

		if (pipd->status < 0) {
			return (pipd->status);
		}

		len = pipd->actual_length;
		pipd->actual_length = 0;
		src = urb->transfer_buffer + pipd->offset;

		if (src != dst) {
			// Need to copy since isoc buffers not full
			while (len--) {
				*dst++ = *src++;
			}
		} else {
			// No need to copy, just update destination buffer
			dst += len;
		}
	}
	// Return size of flattened buffer
	return (dst - (unsigned char *)urb->transfer_buffer);
}

// ------------------------------------------------------------------------- //

int main(int argc, char *argv[]) {

    if (argc != 2) {
        usage();
        return 1;
    }

    int opt = atoi(argv[1]);
    switch(opt) {

    // big-arr
    case 0:
    {
          int _len_urb0 = 65025;
          struct urb * urb = (struct urb *) malloc(_len_urb0*sizeof(struct urb));
          for(int _i0 = 0; _i0 < _len_urb0; _i0++) {
              urb[_i0].status = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].number_of_packets = ((-2 * (next_i()%2)) + 1) * next_i();
          int _len_urb__i0__transfer_buffer0 = 1;
          urb[_i0].transfer_buffer = (unsigned char *) malloc(_len_urb__i0__transfer_buffer0*sizeof(unsigned char));
          for(int _j0 = 0; _j0 < _len_urb__i0__transfer_buffer0; _j0++) {
            urb[_i0].transfer_buffer[_j0] = ((-2 * (next_i()%2)) + 1) * next_i();
          }
          int _len_urb__i0__iso_frame_desc0 = 1;
          urb[_i0].iso_frame_desc = (struct usb_iso_packet_descriptor *) malloc(_len_urb__i0__iso_frame_desc0*sizeof(struct usb_iso_packet_descriptor));
          for(int _j0 = 0; _j0 < _len_urb__i0__iso_frame_desc0; _j0++) {
              urb[_i0].iso_frame_desc->status = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].iso_frame_desc->actual_length = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].iso_frame_desc->offset = ((-2 * (next_i()%2)) + 1) * next_i();
        
          }
        
          }
        
          int benchRet = st5481_isoc_flatten(urb);
          printf("%d\n", benchRet); 
          for(int _aux = 0; _aux < _len_urb0; _aux++) {
          free(urb[_aux].transfer_buffer);
          }
          for(int _aux = 0; _aux < _len_urb0; _aux++) {
          free(urb[_aux].iso_frame_desc);
          }
          free(urb);
        
        break;
    }
    // big-arr-10x
    case 1:
    {
          int _len_urb0 = 100;
          struct urb * urb = (struct urb *) malloc(_len_urb0*sizeof(struct urb));
          for(int _i0 = 0; _i0 < _len_urb0; _i0++) {
              urb[_i0].status = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].number_of_packets = ((-2 * (next_i()%2)) + 1) * next_i();
          int _len_urb__i0__transfer_buffer0 = 1;
          urb[_i0].transfer_buffer = (unsigned char *) malloc(_len_urb__i0__transfer_buffer0*sizeof(unsigned char));
          for(int _j0 = 0; _j0 < _len_urb__i0__transfer_buffer0; _j0++) {
            urb[_i0].transfer_buffer[_j0] = ((-2 * (next_i()%2)) + 1) * next_i();
          }
          int _len_urb__i0__iso_frame_desc0 = 1;
          urb[_i0].iso_frame_desc = (struct usb_iso_packet_descriptor *) malloc(_len_urb__i0__iso_frame_desc0*sizeof(struct usb_iso_packet_descriptor));
          for(int _j0 = 0; _j0 < _len_urb__i0__iso_frame_desc0; _j0++) {
              urb[_i0].iso_frame_desc->status = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].iso_frame_desc->actual_length = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].iso_frame_desc->offset = ((-2 * (next_i()%2)) + 1) * next_i();
        
          }
        
          }
        
          int benchRet = st5481_isoc_flatten(urb);
          printf("%d\n", benchRet); 
          for(int _aux = 0; _aux < _len_urb0; _aux++) {
          free(urb[_aux].transfer_buffer);
          }
          for(int _aux = 0; _aux < _len_urb0; _aux++) {
          free(urb[_aux].iso_frame_desc);
          }
          free(urb);
        
        break;
    }
    // empty
    case 2:
    {
          int _len_urb0 = 1;
          struct urb * urb = (struct urb *) malloc(_len_urb0*sizeof(struct urb));
          for(int _i0 = 0; _i0 < _len_urb0; _i0++) {
              urb[_i0].status = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].number_of_packets = ((-2 * (next_i()%2)) + 1) * next_i();
          int _len_urb__i0__transfer_buffer0 = 1;
          urb[_i0].transfer_buffer = (unsigned char *) malloc(_len_urb__i0__transfer_buffer0*sizeof(unsigned char));
          for(int _j0 = 0; _j0 < _len_urb__i0__transfer_buffer0; _j0++) {
            urb[_i0].transfer_buffer[_j0] = ((-2 * (next_i()%2)) + 1) * next_i();
          }
          int _len_urb__i0__iso_frame_desc0 = 1;
          urb[_i0].iso_frame_desc = (struct usb_iso_packet_descriptor *) malloc(_len_urb__i0__iso_frame_desc0*sizeof(struct usb_iso_packet_descriptor));
          for(int _j0 = 0; _j0 < _len_urb__i0__iso_frame_desc0; _j0++) {
              urb[_i0].iso_frame_desc->status = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].iso_frame_desc->actual_length = ((-2 * (next_i()%2)) + 1) * next_i();
          urb[_i0].iso_frame_desc->offset = ((-2 * (next_i()%2)) + 1) * next_i();
        
          }
        
          }
        
          int benchRet = st5481_isoc_flatten(urb);
          printf("%d\n", benchRet); 
          for(int _aux = 0; _aux < _len_urb0; _aux++) {
          free(urb[_aux].transfer_buffer);
          }
          for(int _aux = 0; _aux < _len_urb0; _aux++) {
          free(urb[_aux].iso_frame_desc);
          }
          free(urb);
        
        break;
    }
    default:
        usage();
        break;

    }

    return 0;
}
